package cn.wenhaha.common.db;

import cn.hutool.db.ds.simple.SimpleDataSource;
import cn.hutool.system.SystemUtil;
import cn.wenhaha.common.config.Global;
import com.jfinal.plugin.activerecord.dialect.Sqlite3Dialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.activerecord.generator.TypeMapping;

import javax.sql.DataSource;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Date;

/**
 * 代码生成
 * --------
 *
 * @author ：wyndem
 * @Date ：Created in 2022-03-19 21:02
 */
public class _Generator {

    /**
     * 在此统一添加不参与生成的 table。不参与生成的 table 主要有：
     * 1：用于建立表之间关系的关联表，如 account_role
     * 2：部分功能使用 Db + Record 模式实现，所涉及到的 table 也不参与生成
     */
    private static String[] whitelTable = {
            "sys_data_source","sys_mapping","sys_sync_plugin","sys_user","sys_sync_error","sys_sync_report"
    };

    public static DataSource getDataSource() {
//        DataSource ds = new SimpleDataSource("jdbc:mysql://106.52.123.101:3306/demo1?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&serverTimezone=GMT%2B8",
//                "root", "Wen#1234");
//
        DataSource ds = new SimpleDataSource("jdbc:sqlite:"+ Global.getRootPath() +"\\sync.db",
                "", "");
        return ds;
    }

    public static void main(String[] args) {
        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = "cn.wenhaha.common.db.bean";

        // base model 所使用的包名
        String baseModelPackageName = modelPackageName + ".model";

        // base model 文件保存路径
        String baseModelOutputDir = SystemUtil.getUserInfo().getCurrentDir()+ "\\common"
                + "/src/main/java/" + baseModelPackageName.replace('.', '/');

        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";

        System.out.println("输出路径：" + baseModelOutputDir);

        // 创建生成器
        Generator gen = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);

        // 设置数据库方言
//        gen.setDialect(new MysqlDialect());
        gen.setDialect(new Sqlite3Dialect());

        // 设置是否生成字段备注
        gen.setGenerateRemarks(true);



        gen.addWhitelist(whitelTable);
        // 设置是否在 Model 中生成 dao 对象
        gen.setGenerateDaoInModel(false);

        // 设置是否生成字典文件
        gen.setGenerateDataDictionary(false);


        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
        // gernerator.setRemovedTableNamePrefixes("t_");

        // 将 mysql 8 以及其它原因之下生成 jdk 8 日期类型映射为 java.util.Date，便于兼容老项目，也便于习惯使用 java.util.Date 的同学
        TypeMapping tm = new TypeMapping();
        tm.addMapping(LocalDateTime.class, Date.class);
        tm.addMapping(LocalDate.class, Date.class);
        // tm.addMapping(LocalTime.class, LocalTime.class);		// LocalTime 暂时不变
        gen.setTypeMapping(tm);

        // 生成
        gen.generate();
    }
}
